////////////////////////////////////////////////////////////////////////////////
// Copyright (C) 2008 Michiel Helvensteijn                                    //
//                                                                            //
// This file is part of the Mist compiler.                                    //
//                                                                            //
// The Mist compiler is free software: you can redistribute it and/or modif   //
// it under the terms of the GNU General Public License as published by       //
// the Free Software Foundation, either version 3 of the License, or          //
// (at your option) any later version.                                        //
//                                                                            //
// The Mist compiler is distributed in the hope that it will be useful,       //
// but WITHOUT ANY WARRANTY; without even the implied warranty of             //
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the              //
// GNU General Public License for more details.                               //
//                                                                            //
// You should have received a copy of the GNU General Public License          //
// along with the Mist compiler.  If not, see <http://www.gnu.org/licenses/>. //
////////////////////////////////////////////////////////////////////////////////

#include "visitors/findtemporariesvisitor.hpp"

/******************************************************************************
 * Includes                                                                   *
 ******************************************************************************/

#include <boost/shared_ptr.hpp>
using boost::dynamic_pointer_cast;

#include "ast/nodetypes.hpp"
#include "visitors/expressionnavigator.hpp"

/******************************************************************************
 * DeclarationVisitor1 Class (first pass)                                     *
 ******************************************************************************/

class FindTemporariesVisitor : public ExpressionNavigator {
	public:
		
		FindTemporariesVisitor() : begun(false) {}
		
		void visit(shared_ptr<ComparisonChain> node);
		void visit(shared_ptr<ImpliesChain> node);
		void visit(shared_ptr<FunctionDeclaration> node);
		void visit(shared_ptr<Declarations> node);
		
		vector<shared_ptr<Type> > temporaries();
	
	private:
		
		bool begun;
		
		vector<shared_ptr<Type> > _temporaries;
};

/******************************************************************************
 * Function Implementations                                                   *
 ******************************************************************************/

vector<shared_ptr<Type> > find_temporaries(shared_ptr<FunctionDeclaration> node) {
	assert(node);
	
	shared_ptr<FindTemporariesVisitor> visitor(new FindTemporariesVisitor);
	node->accept(visitor);
	
	return visitor->temporaries();
}

vector<shared_ptr<Type> > find_temporaries(shared_ptr<Declarations> node) {
	assert(node);
	
	shared_ptr<FindTemporariesVisitor> visitor(new FindTemporariesVisitor);
	node->accept(visitor);
	
	return visitor->temporaries();
}

/******************************************************************************/

void FindTemporariesVisitor::visit(shared_ptr<ComparisonChain> node) {
	node->operands()[0]->accept(shared_from_this());
	for (unsigned i = 1; i < node->operand_count() - 1; ++i) {
		if (node->operands()[i]->equals(Null()) == false) {
			_temporaries.push_back(node->operands()[i]->type());
			node->operands()[i]->set_temp_id(_temporaries.size());
		}
		node->operands()[i]->accept(shared_from_this());
	}
	node->operands().back()->accept(shared_from_this());
}

void FindTemporariesVisitor::visit(shared_ptr<ImpliesChain> node) {
	node->operands()[0]->accept(shared_from_this());
	for (unsigned i = 1; i < node->operand_count() - 1; ++i) {
		_temporaries.push_back(node->operands()[i]->type());
		node->operands()[i]->set_temp_id(_temporaries.size());
		node->operands()[i]->accept(shared_from_this());
	}
	node->operands().back()->accept(shared_from_this());
}

void FindTemporariesVisitor::visit(shared_ptr<FunctionDeclaration> node) {
	if (!begun) {
		begun = true;
		StatementNavigator::visit(node);
	}
}

void FindTemporariesVisitor::visit(shared_ptr<Declarations> node) {
	begun = true;
	StatementNavigator::visit(node);
}

vector<shared_ptr<Type> > FindTemporariesVisitor::temporaries() {
	return _temporaries;
}
